Aligning LLMs with Sparse Retrieval via Adaptive Sparse Supervision
自适应稀疏监督下的LLM稀疏检索对齐
🧭 LLM稀疏召回 · 自适应稀疏对齐 ★ 深度解读
Soochow University
用动态α-truncation对齐SPLADE词表,1B LLM稀疏检索在L0=182时接近Lion-sp效果且延迟降11倍
📌 工作参考价值(4 点)
- 在搜索/广告召回中若要用LLM做稀疏召回,可复用「投影到SPLADE词表 + max-pooling + log(1+ReLU)」结构,用1B模型+LoRA即可与倒排索引兼容,避免在线生成式开销
- 训练阶段用SPLADE作为anchor,对每个输入按累积激活质量动态选取监督support(α-truncation),比固定Top-K更稳:在消融中MRR@10从α-mean的0.303提升到0.322
- 两阶段训练值得借鉴:先做稀疏对齐预训练,再做对比学习+KL蒸馏+FLOPS正则,可让索引条目减少40.7%,同时MRR@10从0.3619提升到0.3998
- 推理时无需额外truncation,只靠训练获得的稀疏性;对query和document分别设置FLOPS正则系数,可把L0从1170压到182,避免83GB索引和>1s查询延迟